Stock-Analysis

StockAnalysis

This program empowers users to analyze stock data and compare its price trends over various timeframes: today, yesterday, last month, and last year. The program offers the flexibility to input specific stock tickers or retrieve ‘n’ stocks based on the S&P index. Usage

The program provides insights into stock performance and visualizes trends. Users can input stock tickers or opt for automatic retrieval based on the S&P index. Sample Output

For instance, the program can analyze the performance of AAPL over the past day, month, and year, plotting the trends for visual understanding.

image

Changes Made

The program underwent significant enhancements:

1. Input Validation: Implemented robust user input validation for seamless data retrieval.
2. Data Fetching Enhancement: Utilized the Yahoo Finance API to fetch accurate stock data.
3. Improved Data Plotting: Enhanced data plotting with titles and labels for clearer visualization.
4. Handling No Data: Addressed scenarios where no data is available for specific stocks, ensuring smoother execution.
5. Code Organization: Structured the code for better organization and readability.

Source Code: StockAnalysis.py

import yfinance as yf
import pandas as pd
from datetime import datetime, timedelta
import matplotlib.pyplot as plt

# Function to get a list of tickers for analysis
def get_ticker_list():
    while True:
        try:
            option = int(input("You have 2 options:\n1. Analyze top stocks in S&P index\n2. Specific Stocks\nEnter option (1/2): "))

            if option == 1:
                stocknum = int(input("How many stocks do you want to analyze? Enter a number: "))
                toptickers = pd.read_html('https://en.wikipedia.org/wiki/List_of_S%26P_500_companies')[0]
                return toptickers.head(stocknum).Symbol.to_list()
            elif option == 2:
                inputstr = input("Enter comma-separated ticker symbols of interested stocks: ")
                return inputstr.upper().split(",")
            else:
                print("Please enter a valid option (1/2).")
        except ValueError:
            print("Please input a valid integer only.")

# Function to get the time period for analysis
def get_time_period():
    try:
        start_date = input("Enter the start date (YYYY-MM-DD): ")
        end_date = input("Enter the end date (YYYY-MM-DD): ")
        return start_date, end_date
    except ValueError:
        print("Invalid date format. Please use the format YYYY-MM-DD.")

# Main function
def main():
    tickerlist = get_ticker_list()
    
    start_date, end_date = get_time_period()  # Get time period for analysis
    
    data = pd.DataFrame()

    # Loop through each ticker to fetch data
    for ticker in tickerlist:
        try:
            y = yf.Ticker(ticker)
            df = y.history(period="1d", start=start_date, end=end_date)
            if not df.empty:
                data[ticker] = df["Close"]
            else:
                print("No data found for", ticker)
        except Exception as e:
            print("An error occurred while fetching data for", ticker, ":", str(e))

    if not data.empty:
        # Plotting the data
        data.plot(title="Stock Price Analysis")
        plt.xlabel("Date")
        plt.ylabel("Price")
        plt.show()
        
        print(data.head())
    else:
        print("No data available for analysis.")

# Entry point of the program
if __name__ == "__main__":
    main()